package cn.edu.xjtu.work.removeElements;

import cn.edu.xjtu.Util.ListNode.ListNode;
import cn.edu.xjtu.Util.ListNode.ListNodeUtil;

/**
 * 203. 移除链表元素
 * 
 * 给你一个链表的头节点 head 和一个整数 val ，请你删除链表中所有满足 Node.val == val 的节点，并返回 新的头节点 。
 */
public class Solution {
  public static void main(String[] args) {
    ListNode head = ListNodeUtil.generate(new int[] { 1, 2, 6, 3, 4, 5, 6 });
    ListNodeUtil.print(head);
  }

  /**
   * 添加虚拟头节点
   * 
   * @param head
   * @param val
   * @return
   */
  public ListNode removeElements(ListNode head, int val) {
    if (head == null) {
      return null;
    }
    ListNode newHead = new ListNode(-1, head);
    head = newHead;
    while (head.next != null) {
      if (head.next.val == val) {
        head.next = head.next.next;
        continue;
      }
      head = head.next;
      if (newHead.val == -1) {
        newHead = head;
      }
    }
    return newHead.val == -1 ? null : newHead;
  }

  /**
   * 递归实现
   * 
   * @param head
   * @param val
   * @return
   */
  public ListNode removeElements1(ListNode head, int val) {
    if (head == null) {
      return null;
    }
    head.next = removeElements1(head.next, val);
    return head.val == val ? head.next : head;
  }
}
